from wordcloud import WordCloud, STOPWORDS
import matplotlib.pyplot as plt
import numpy as np
from PIL import Image

with open("./status/sample.txt", encoding='utf-8') as f:
    text = f.read()

mask = np.array(Image.open("./status/heihei.png"))
# mask = np.array(Image.open("./status/haha.png"))

# 1、font_path：字体文件路径。
# 2、mask：词云的形状。
# 3、background_color：背景颜色。
# 4、max_words：要显示的最大词数。
# 5、stopwords：忽略词列表。
# 6、mode：选择模式。

wordcloud = WordCloud(
    font_path="msyh.ttc",
    stopwords=STOPWORDS.add("said"),
    # background_color="white",
    background_color="rgba(255, 255, 255, 0)",
    max_words=500,
    mask=mask,
    mode="RGBA"
).generate(text)

wordcloud.to_file("./status/output2.png")

# plt.imshow(wordcloud, interpolation="bilinear")
# plt.axis("off")
# plt.show()